absl-py
numpy>=1.18.4
six>=1.14.0
tensorflow>=2.2.0
tensorflow_datasets>=3.1.0
tensorflow_probability>=0.7.0
